import os
import random

# 创建文件夹路径
input_dir = r'C:\Users\ASUS\Desktop\2024年新生赛出题\2024年新生赛出题(非大一组)\G题\G题_input'
os.makedirs(input_dir, exist_ok=True)

# 设置生成数据的参数
N_min, N_max = 930, 1000  # 使 N 相对较小来限制暴力算法可行性
value_min, value_max = 1, 1000

# 随机生成满足条件的测试数据
def generate_test_case(N):
    # 创建一个数组，包含符合题目要求范围的整数
    nums = [random.randint(value_min, value_max) for _ in range(N)]
    return nums

# 生成15组测试数据文件
for idx in range(4, 16):
    # 随机生成数组长度，避免暴力算法能顺利通过
    N = random.randint(N_min, N_max)
    nums = generate_test_case(N)
    
    # 保存到文件
    input_path = os.path.join(input_dir, f'{idx}.in')
    with open(input_path, 'w') as f:
        f.write(f"{N}\n")
        f.write(" ".join(map(str, nums)) + "\n")

print("15组测试数据已生成并保存在指定目录中。")
